from datasets.datasets import DatasetType
from models.losses.loss_functions import LossFunctionType
from models.optimizers.optimizers import OptimizerType
from models.models import ModelType


config = dict(
    model = ModelType.DEEPLAB_V3,
    num_classes = 19,

    dataset = DatasetType.CITYSCAPES,
    input_size = [768, 768],
    batch_size = 4,
    data_root = './datasets/cityscape',

    optimizer = OptimizerType.SGD,
    lr = 0.01,
    momentum = 0.9,
    weight_decay = 0.0001,

    loss_function = LossFunctionType.Cross_Entropy,

    epochs = 150,
)
